package com.ziji.rocketmq.order;

import org.apache.rocketmq.client.exception.MQBrokerException;
import org.apache.rocketmq.client.exception.MQClientException;
import org.apache.rocketmq.client.producer.DefaultMQProducer;
import org.apache.rocketmq.client.producer.MessageQueueSelector;
import org.apache.rocketmq.client.producer.SendResult;
import org.apache.rocketmq.common.message.Message;
import org.apache.rocketmq.common.message.MessageQueue;
import org.apache.rocketmq.remoting.exception.RemotingException;

import java.util.List;

/**
 * 顺序消息
 *
 * @author <a href="mailto:ziji.zhao@aliyun.com">ziji</a>
 * @since 2021/11/12
 */
public class OrderedProducer {
    public static void main(String[] args) throws MQClientException, MQBrokerException, RemotingException, InterruptedException {
        // 创建一个 producer，参数为 Producer Group 名称
        DefaultMQProducer producer = new DefaultMQProducer("pg");
        // 指定 NameServer 地址
//        producer.setNamesrvAddr("rocketmqOS:9876");
        producer.setNamesrvAddr("localhost:9876");

        // 开启生产者
        producer.start();

        // 生产并发送 100 条消息
        for (int i=0; i<100; i++) {
            // 为了演示简单，使用整形数作为 orderId
            Integer orderId = i;
            byte[] body = ("Hi," + i).getBytes();
            Message msg = new Message("orderedTopic", "orderedTag", body);
//            Message msg = new Message("someTopic", "someTag" + (i % 3), body);
            // 为消息指定 key
            msg.setKeys("key-" + i);

            // DefaultMQProducer#send(Message msg, MessageQueueSelector selector, Object arg)
            // 的第三个参数值会传递给选择器的 select() 的第三个参数
            SendResult sendResult = producer.send(msg, new MessageQueueSelector() {
                // 具体的选择算法在该方法中定义
                @Override
                public MessageQueue select(List<MessageQueue> mqs, Message msg, Object arg) {
                    String keys = msg.getKeys(); // 除了可以通过 arg 传递关键参数，还可以通过 key 来传递关键参数

                    Integer id = (Integer) arg; // orderId
                    int index = id % mqs.size();
                    return mqs.get(index);      // 通过 orderId 按 queue 数量取模来达到"同 orderId 的消息分区一致"的目的
                }
            }, orderId);
            System.out.println(sendResult);
        }
        // 关闭 producer
        producer.shutdown();
    }
}
